package cn.kinoko.service.common.event.component;

import cn.kinoko.common.utils.CompletableFutureUtil;
import cn.kinoko.service.common.event.EventHandler;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.ExecutorService;
import java.util.stream.Collectors;

/**
 * 事件分发处理器
 *
 * @author kinoko
 */
@Slf4j
@Component
public class EventDistributeHandler {

    public static final List<Event> EVENT_TRACKING_QUEUE = new CopyOnWriteArrayList<>();

    @Resource
    private Map<Integer, EventHandler> eventHandlerMap;
    @Resource
    private ExecutorService executorService;

    /**
     * 监听事件并分发
     */
    public synchronized void listenAndDistributeEvent() {
        // 从缓存的事件缓冲队列中拉取事件
        if (!EVENT_TRACKING_QUEUE.isEmpty()) {
            // 获取当前事件数量
            int eventCount = EVENT_TRACKING_QUEUE.size();
            log.info("[事件分发处理器] 本次分发事件总数：{}", eventCount);
            List<Event> eventList = EVENT_TRACKING_QUEUE.subList(0, eventCount);
            // 根据事件类型分组，用于后续事件派发
            Map<Integer, List<Event>> eventTrackingMap = eventList.stream().collect(Collectors.groupingBy(event -> event.getEventTopic().getCode()));
            // 事件分发
            try {
                // 分发处理事件（异步提高消费效率）
                CompletableFutureUtil.runAsyncForEach(eventTrackingMap.entrySet(), entry -> {
                    try {
                        Integer eventType = entry.getKey();
                        List<Event> eventQueue = entry.getValue();
                        // 从事件处理类加载器中获取加载的处理类
                        EventHandler handler = eventHandlerMap.get(eventType);
                        if (handler != null) {
                            // 消费事件
                            handler.eventHandle(eventQueue);
                        }
                    } catch (Exception e) {
                        log.error("[事件分发处理器] 异步消费事件捕获异常：", e);
                    }
                }, executorService);
            } catch (Exception e) {
                log.error("[事件分发处理器] 异步消费事件捕获异常：", e);
            }
            // 移除已消费的事件
            eventList.clear();
            log.info("[事件分发处理器] 分发完毕");
        }
    }

}
